package com.zjj.algorithm.learning.leetcode.linkedlist;

/**
 * 2. 两数相加 中档题
 * <p>
 * 给你两个 非空 的链表，表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的，并且每个节点只能存储 一位 数字。
 * <p>
 * 请你将两个数相加，并以相同形式返回一个表示和的链表。
 * <p>
 * 你可以假设除了数字 0 之外，这两个数都不会以 0 开头。
 * <p>
 * <p>
 * <p>
 * 示例 1：
 * <p>
 * <p>
 * 输入：l1 = [2,4,3], l2 = [5,6,4]
 * 输出：[7,0,8]
 * 解释：342 + 465 = 807.
 * 示例 2：
 * <p>
 * 输入：l1 = [0], l2 = [0]
 * 输出：[0]
 * 示例 3：
 * <p>
 * 输入：l1 = [9,9,9,9,9,9,9], l2 = [9,9,9,9]
 * 输出：[8,9,9,9,0,0,0,1]
 * <p>
 * <p>
 * 提示：
 * <p>
 * 每个链表中的节点数在范围 [1, 100] 内
 * 0 <= Node.val <= 9
 * 题目数据保证列表表示的数字不含前导零
 *
 * @author zjj_admin
 * @date 2022/12/26 14:38
 */
public class AddTwoNumbers {

    public static void main(String[] args) {
        ListNode node1 = new ListNode(1);
        ListNode node2 = new ListNode(2);
        ListNode node3 = new ListNode(3);
        ListNode node4 = new ListNode(4);
        ListNode node5 = new ListNode(5);
        ListNode node6 = new ListNode(6);
        node1.next = node2;
        node2.next = node3;
        node3.next = node4;
        node4.next = node5;
        node5.next = node6;
        addTwoNumbers(node1, node3);
    }

    /**
     * 时间
     * 1 ms
     * 击败
     * 100%
     * 内存
     * 41.4 MB
     * 击败
     * 58.80%
     *
     * @param l1
     * @param l2
     * @return
     */
    public static ListNode addTwoNumbers(ListNode l1, ListNode l2) {
        int len1 = getLength(l1);
        int len2 = getLength(l2);
        int len = Math.max(len1, len2);
        int[] nums1 = new int[len];
        int[] nums2 = new int[len];

        int index1 = len - 1, index2 = len - 1;
        ListNode curr1 = l1;
        while (curr1 != null) {
            nums1[index1--] = curr1.val;
            curr1 = curr1.next;
        }
        ListNode curr2 = l2;
        while (curr2 != null) {
            nums2[index2--] = curr2.val;
            curr2 = curr2.next;
        }
        int[] nums = new int[len + 1];
        int sup = 0;
        for (int i = len - 1; i >= 0; i--) {
            nums[i + 1] = (nums1[i] + nums2[i] + sup) % 10;
            sup = (nums1[i] + nums2[i] + sup) >= 10 ? 1 : 0;
        }
        if (sup > 0) {
            nums[0] = 1;
        }
        ListNode head = new ListNode();
        ListNode curr = new ListNode();
        for (int i = nums.length - 1; i >= 0; i--) {
            if (i == 0 && nums[i] == 0) {
                continue;
            }
            if (i == nums.length - 1) {
                head = new ListNode(nums[i]);
                curr = head;
            } else {
                curr.next = new ListNode(nums[i]);
                curr = curr.next;
            }
        }
        return head;
    }

    /**
     * 获取链表的长度
     *
     * @param head
     * @return
     */
    public static int getLength(ListNode head) {
        int res = 0;
        if (head == null) {
            return res;
        }
        ListNode curr = head;
        while (curr != null) {
            res++;
            curr = curr.next;
        }
        return res;
    }
}
